package com.yhy.wxzn.common.utils;

import java.math.BigDecimal;

/**
 * 浮点数数学运算帮助类
 *
 */
public class DoubleUtils {

    /** 默认精度 */
    private static final int DEF_DIV_SCALE = 10;

    /**
     * 按精度格式化
     */
    public static double formatByPrecision(Object o, int precision) {
        BigDecimal bd = new BigDecimal(o.toString());
        BigDecimal bdWithPrecision = bd.setScale(precision, BigDecimal.ROUND_HALF_UP);
        return bdWithPrecision.doubleValue();
    }

    /**
     * 比较大小
     *
     * <pre>
     * eg. source < target >>>> -1
     * eg. source > target >>>> 1
     * eg. source == target >>>> 0
     * </pre>
     */
    public static int compareByPrecision(Double source, Double target, int precision) {
        BigDecimal src = (new BigDecimal(source.toString())).setScale(precision, BigDecimal.ROUND_HALF_UP);
        BigDecimal dst = (new BigDecimal(target.toString())).setScale(precision, BigDecimal.ROUND_HALF_UP);
        return src.compareTo(dst);
    }

    /**
     * 提供精确的加法运算。
     *
     * @param v1 被加数
     * @param v2 加数
     * @return 两个参数的和
     */
    public static double add(double v1, double v2, int scale) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.add(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供精确的加法运算。
     *
     * @param v1 被加数
     * @param v2 加数
     * @return 两个参数的和
     */
    public static double add(double v1, double v2) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.add(b2).setScale(BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供精确的减法运算。
     *
     * @param v1 被减数
     * @param v2 减数
     * @return 两个参数的差
     */
    public static double sub(double v1, double v2, int scale) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.subtract(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 减法，默认精度
     *
     * @param v1 减数
     * @param v2 被减数
     */
    public static double sub(double v1, double v2) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.subtract(b2).setScale(BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 乘法，指定精度
     *
     * @param v1 乘数
     * @param v2 被乘数
     */
    public static double mul(double v1, double v2, int scale) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.multiply(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 除法，默认 10 位小数
     *
     * @param v1 除数
     * @param v2 被除数
     */
    public static double div(double v1, double v2) {
        return div(v1, v2, DEF_DIV_SCALE);
    }

    /**
     * 除法，指定精度
     *
     * @param v1 除数
     * @param v2 被除数
     * @param scale 精度
     */
    public static double div(double v1, double v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.divide(b2, scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 将可能为空的 Double 转为 0D
     */
    public static Double toDouble(Object o) {
        if (o == null) {
            return 0D;
        }
        return Double.valueOf(o.toString());
    }
}
